package TreeSet;

import java.util.Comparator;
import java.util.TreeSet;

public class demo4 {
    public static void main(String[] args) {
        //方法二:比较器排序
        //1.创建集合
        //2.o1表示当前要添加的元素
        //3.o2表示已经在红黑树存在的元素
        //返回值规则跟之前一样
        TreeSet<String> ts = new TreeSet<>(new Comparator<String>() {
            @Override
            /*
            需求：请自行选择比较器排序和自然排序两种方式；
            要求：存入四个字符串， “c”, “ab”, “df”, “qwer”
            按照长度排序，如果一样长则按照首字母排序

            采取第二种排序方式：比较器排序
             */
            public int compare(String o1, String o2) {
                //按照长度来
                int tmp = o1.length() - o2.length();
                //如果一样长则按照首字母排序
                    tmp = tmp == 0 ? o1.compareTo(o2) : tmp;
                return tmp;
            }
        });
        //2.添加元素
        ts.add("c");
        ts.add("ab");
        ts.add("df");
        ts.add("qwer");
        System.out.println(ts);
    }
}
